import pygame
from pygame.sprite import Sprite

class Ship(Sprite):
    def __init__(self, ai_settings, screen):
        super(Ship, self).__init__()
        # 初始化飞机并设置初始位置
        self.screen = screen
        self.ai_settings = ai_settings

        # 加载图像并获取矩形
        self.image = pygame.image.load('../image/ship.png')
        self.rect = self.image.get_rect()
        self.screen_rect = screen.get_rect()

        # 飞船位置在底部正中央
        self.rect.centerx = self.screen_rect.centerx
        self.rect.bottom = self.screen_rect.bottom

        # 飞船的属性center中存储
        self.center = float(self.rect.centerx)

        # 移动标志
        self.moving_right = False
        self.moving_left = False

    def blitme(self):
        # 绘制飞船
        self.screen.blit(self.image, self.rect)

    def update(self):
        if self.moving_right and self.rect.right < self.screen_rect.right:
            self.center += self.ai_settings.ship_speed_factor
        if self.moving_left and self.rect.left > 0:
            self.center -= self.ai_settings.ship_speed_factor

        # 根据 self.center 更新 rect 对象
        self.rect.centerx = self.center

    # 重置飞船位置
    def center_ship(self):
        self.center = self.screen_rect.centerx